package com.okbihuo.perm.component.file.domain.type;

import com.google.common.collect.Lists;

/**
 * @author xiang
 */

public enum FileType {
    /**
     * 图片
     */
    image,
    /**
     * 视频
     */
    video,
    /**
     * 音频
     */
    audio,
    /**
     * 普通文件
     */
    normal,
    /**
     * 其他文件
     */
    other;

    /**
     * 根据后缀判断类型
     * @param suffix 后缀, 不带点
     * @return 判断的文件类型
     */
    public static FileType checkType(String suffix) {
        suffix = suffix.toLowerCase();
        if(Lists.newArrayList("png","jpg","jpeg","gif","svg","bmp").contains(suffix)) {
            return image;
        }else if (Lists.newArrayList("avi", "mov", "wmv", "ra", "rb", "rmmvb", "mp4").contains(suffix)) {
            return video;
        }else if(Lists.newArrayList("wav","mp3","flac").contains(suffix)) {
            return audio;
        }else {
            return other;
        }
    }
}
